"""
 @Author: Ajian
 @Email: ajian100813@126.com
 @FileName: custom_exception_handler.py
 @DateTime: 2025/1/14 23:01
 @Project: python-rpa
 @Version: 1.0
"""

import sys
import time
import subprocess

def custom_exception_handler(type, value, traceback):
    print("程序异常退出，执行清理操作")
    # 打印异常信息
    sys.__excepthook__(type, value, traceback)

    # 在10秒后重启系统
    print("系统将在10秒后重启...")
    time.sleep(10)
    subprocess.run(["shutdown", "/r", "/t", "0"])

def main():
    # 设置自定义异常处理函数
    sys.excepthook = custom_exception_handler

    # 主程序逻辑
    while True:
        print("运行中...")
        time.sleep(10)
        # 故意引发一个异常以测试
        raise ValueError("测试异常")

if __name__ == "__main__":
    main()

